# 导入工具包
import torch
import torch.nn as nn
'''nn.RNN类初始化主要参数解释:

input_size: 输入张量x中特征维度的大小.
hidden_size: 隐层张量h中特征维度的大小.
num_layers: 隐含层的数量.
nonlinearity: 激活函数的选择, 默认是tanh.'''
rnn = nn.RNN(5, 6, 1)
'''nn.GRU类实例化对象主要参数解释:
input: 输入张量x.
h0: 初始化的隐层张量h.'''
input = torch.randn(1, 3, 5)
h0 = torch.randn(1, 3, 6)
output, hn = rnn(input, h0)



